import mplfinance as mpf
import pandas as pd
import akshare as ak

print("starting mplfinance .....")
df = ak.stock_zh_a_hist(
    symbol="000001",
    period="daily",
    start_date="20230601",
    end_date="20231101",
    adjust="",
)
df = df.iloc[:, 0:6]
df.日期 = pd.to_datetime(df.日期)

# 列名记得这样定义好
df.columns = ["Date", "Open", "Close", "High", "Low", "Volume"]
df.set_index("Date", inplace=True)

# mpf.plot(df, type="candle", mav=(5, 10, 20), volume=True)

# 设置样式
mpf.available_styles()

# mpf.plot(
#     df, type="candle", mav=(5, 8, 13), volume=True, style="classic", title="classic"
# )

# mpf.plot(
#     df, type="candle", mav=(5, 8, 13), volume=True, style="charles", title="charles"
# )

# mpf.plot(df, type="candle", mav=(5, 8, 13), volume=True, style="mike", title="mike")

# mpf.plot(
#     df,
#     type="candle",
#     mav=(5, 8, 13),
#     volume=True,
#     style="starsandstripes",
#     title="starsandstripes",
# )

mpf.plot(df, type="candle", mav=(5, 8, 13), volume=True, style="yahoo", title="yahoo")

print("end")
